"use client";
import { useRouter } from "next/navigation";
import { useState } from "react";
import { useToast } from "@/components/Toast";
import { type ApiError, api } from "@/lib/client/api";

export function RunActions({
  runId,
  status,
  paperId,
}: {
  runId: string;
  status: string;
  paperId: string | null;
}) {
  const router = useRouter();
  const toast = useToast();
  const [busy, setBusy] = useState(false);
  const act = async (path: string, body: unknown, ok: string) => {
    setBusy(true);
    try {
      await api(path, { json: body });
      toast.push("success", ok);
      router.refresh();
    } catch (e) {
      toast.push("error", (e as ApiError).message);
    } finally {
      setBusy(false);
    }
  };
  const canCancel = !["published", "cancelled", "failed", "cancelling"].includes(status);
  const canRetry = ["failed", "quality_failed"].includes(status);
  const canPublish = ["awaiting_approval", "ready_to_publish"].includes(status);
  return (
    <div className="flex flex-wrap gap-2">
      {canPublish && (
        <button
          type="button"
          className="btn btn-primary"
          disabled={busy}
          onClick={() =>
            confirm("Publish this run as a public working paper?") &&
            act("/api/admin/publications", { runId }, "Publication started.")
          }
        >
          Publish
        </button>
      )}
      {canRetry && (
        <button
          type="button"
          className="btn"
          disabled={busy}
          onClick={() => act(`/api/admin/runs/${runId}/retry`, {}, "Retry queued from the last checkpoint.")}
        >
          Retry from checkpoint
        </button>
      )}
      {canCancel && (
        <button
          type="button"
          className="btn btn-danger"
          disabled={busy}
          onClick={() =>
            confirm("Cancel this run? Logs and the frozen input are preserved.") &&
            act(`/api/admin/runs/${runId}/cancel`, {}, "Cancellation requested.")
          }
        >
          Cancel
        </button>
      )}
      {paperId && (
        <a className="btn" href="/admin/publications">
          Published paper
        </a>
      )}
    </div>
  );
}